Removing elements from a list and modifying another list in Python